using Aurora.AI.MetaGLM.Models.RequestModels.WebSearchModels;

namespace Aurora.AI.MetaGLM.Models.RequestModels.FunctionModels
{
    public class FunctionTool
    {
        public string type { get; set; } = "function";
        public Dictionary<string, object> function { get; set; } = new();
        public WebSearchTool web_search { get; set; } = new();
        public FunctionTool SetName(string name)
        {
            function["name"] = name;
            return this;
        }

        public FunctionTool SetDescription(string desc)
        {
            function["description"] = desc;
            return this;
        }

        public FunctionTool SetParameters(FunctionParameters param)
        {
            function["parameters"] = param;
            return this;
        }
    }
}